import pandas as pd
from argostranslate import package, translate

# 安装 Argos Translate 的模型（如果尚未安装）
# package.install_from_path('path_to_your_model.argosmodel')  # 请替换为你的模型路径

# 加载 Argos Translate 的模型
installed_languages = translate.get_installed_languages()
print(installed_languages)
# 假设你已经安装了英文到中文的翻译模型
from_lang = "en"  # 英文
to_lang = "zh"     # 中文

# 读取数据
data = pd.read_csv('output.csv')

# 翻译Word列的英语到中文，并存入chinese列
data['chinese'] = data['Word'].apply(lambda x: print(translate.translate(x, from_lang, to_lang)) or translate.translate(x, from_lang, to_lang))

# print(translate.translate("Hello", from_lang, to_lang))
# 保存结果
data.to_csv('output_tro.csv', index=False, encoding='utf-8-sig')
